import pymysql

# 2) 使用 PyMySQL 库建立与 MySQL 的连接，并返回一个 db 对象。cormect( )方法中的“localhost" 参数为主机名"，"root" 参数为连接 MySQL 的用户名，"密码" 参数为 MySQL
# 的连接密码，"test" 参数表示操作的 MySQL 数据库。
db = pymysql.connect(host='localhost', user='root', password='root', port=3306, database='test')

# 3) 使用 connector 对象的 cursor( )方法建立对 MySQL 的操作游标。
cursor = db.cursor()

# 4) 使用字符串编写完成 SQL 语句。注意，这里使用三引号表示多行字符串。这里的SQL 语句使用INSERT INTO 向 employee 表的 first_name、last_name、age、sex 和 income
# 字段中分别插入 VALUES 为 '5','20','M'和 5000 的值。

sql = """
INSERT INTO
  EMPLOYEE (FIRST_NAME, LAST_NAME, AGE, SEX, INCOME)
VALUES
  ('Mac', 'Su', 20, 'M', 5000)
"""
# 5) 使用 try 和 except 语句执行游标 cursor 的 SQL 语句，并使用 commit( ) 方法提交至MySQL 数据库服务器，rollback。方法表示如果在整个提交过程中出现任何问题，则实现事务性的回滚操作。

try:
    cursor.execute(sql)
    db.commit()
except:
    db.rollback()

# 6) 关闭数据库连接。
db.close()